How to make tree using Python